package config

import (
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"os"
)

/*
	用于初始化数据库连接的函数 InitDB
*/

var DB *gorm.DB

// GetDatabaseDSN 从环境变量获取数据库配置（带默认值）
func GetDatabaseDSN() string {
	dsn := os.Getenv("DB_DSN")
	if dsn == "" {
		// 默认开发环境配置
		return "root:123456@tcp(localhost:3306)/library?charset=utf8mb4&parseTime=True&loc=Local"
	}
	return dsn
}

// InitDB 初始化数据库连接（保持原有实现）
func InitDB(dsn string) error {
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		return err
	}
	DB = db
	return nil
}
